
in_params=[]
in_params.append({"bm":256, "bn":16, "bk":256, "ydim":8, "xdim":32, "tm":128, "tn":4})
in_params.append({"bm":256, "bn":16, "bk":256, "ydim":8, "xdim":32, "tm":32, "tn":16})
in_params.append({"bm":256, "bn":16, "bk":256, "ydim":8, "xdim":32, "tm":64, "tn":8})
in_params.append({"bm":256, "bn":16, "bk":256, "ydim":8, "xdim":32, "tm":256, "tn":2})

def eval(inpm):
    outpm={"sum_acc_mem":0, "inner_loop":0 }
    bm = inpm["bm"]
    bn = inpm["bn"]
    bk = inpm["bk"]
    tm = inpm["tm"]
    tn = inpm["tn"]
    ydim = inpm["ydim"]
    xdim = inpm["xdim"]
    outpm["sum_acc_mem"] = (tm*bk+bk*tn*32)*8
    outpm["inner_loop"] = tm*tn
    return outpm

for in_param in in_params:
    out_param = eval(in_param)
    print(out_param)
    
    
